# -*- coding: UTF-8 -*-
"""
@File    :   set_config_info.py
@Author  :   AnimateX
@Contact :   animatex@163.com
@License :   Copyright © 2022-2023 RockChips. All rights reserved.

@Create Time           @Software Version
-----------------      -----------------
2023/12/6 17:59        Version_1.0   

------------------------------------------------------------------
@Description :

------------------------------------------------------------------
"""
import os

car_license_det_configs = [
    'path: /home/dataset_b/car_license_data/car_data/images',
    'train: train',
    'val: val',
    'nc: 1',
    'names: ["license_plate"]'
]

DATASET_PATH = '/home/dataset_b/car_license_data/car_data'
YAML_FILE_NAME = 'dataset.yaml'
YAML_FILE_PATH = os.path.join(DATASET_PATH, YAML_FILE_NAME)

#  Write info to .yaml
with open(YAML_FILE_PATH, 'w') as f:
    for config in car_license_det_configs:
        f.write(f'{config}\n')
